package com.vben.ai.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.image.ImageModel;
import org.springframework.ai.image.ImagePrompt;
import org.springframework.ai.image.ImageResponse;
import org.springframework.stereotype.Service;

@Service
@Slf4j
public class ImageService {

    private final ImageModel imageModel;

    public ImageService(ImageModel imageModel) {
        this.imageModel = imageModel;
    }

    public String image(String prompt) {
        ImageResponse imageResponse = imageModel.call(new ImagePrompt(prompt));
        return imageResponse.getResult().getOutput().getUrl();
    }
}
